import torch

torch.device('cpu')
r = torch.rand(5, 5)

print(r)